#!/bin/bash

nmt_data=/home/zrs/nmt_homework/data/iwslt15_en_vt
nmt_model=./nmt_model/iwslt15_en_vt

rm -rf $nmt_model

CUDA_VISIBLE_DEVICES=0,1,2 python -m nmt.nmt \
	--src=vi --tgt=en \
	--vocab_prefix=${nmt_data}/vocab  \
	--train_prefix=${nmt_data}/train \
	--dev_prefix=${nmt_data}/tst2012  \
	--test_prefix=${nmt_data}/tst2013 \
	--out_dir=$nmt_model \
	--num_train_steps=12000 \
	--steps_per_stats=100 \
	--num_layers=2 \
	--num_units=128 \
	--dropout=0.2 \
	--num_gpus 3 \
	--metrics=bleu 
